import torch
from torchtext.vocab import Vocab
from config.Config import BASE_DIR


def get_nl_vocab() -> Vocab:
    return torch.load(BASE_DIR / 'resource' / 'data' / 'consum' / 'nl_vocab.pkl')


def get_code_vocab() -> Vocab:
    return torch.load(BASE_DIR / 'resource' / 'data' / 'consum' / 'code_vocab.pkl')


def get_big_vocab() -> Vocab:
    return torch.load(BASE_DIR / 'resource' / 'data' / 'consum' / 'big_vocab.pkl')


big_vocab = get_big_vocab()

if __name__ == '__main__':
    v = get_big_vocab()
